import selenium
from selenium import webdriver
import os
import time
import pickle   # 记录

# 一，自动登录
    # 判断当前有没有登录


# 大麦网网页
damai_url = 'https://www.damai.cn/'

# 登录页面
login_url = 'https://passport.damai.cn/login?ru=https%3A%2F%2Fwww.damai.cn%2F'

# 抢票页面
target_url = 'https://detail.damai.cn/item.htm?spm=a2oeg.home.card_0.ditem_0.591b23e1Vx6kzp&id=682030358116'

class Concert:
    """初始化加载"""
    def __int__(self):
        self.status = 0     # 状态，标识当前操作执行到了那个步骤
        self.login_method = 1   #{0：模拟登录，1：cookie登录}
        self.driver = webdriver.Chrome()      # 创建浏览器

    """cookies: 登录网站的时候出现的，记录用户信息"""
    def set_cookies(seif):
        seif.driver.get(login_url)
        print("###请扫描登录###")
        time.sleep(10)
        print('###登录成功###')

        # cookie.pkl
        pickle.dump(seif.driver.get_cookie(), open('cookies.pkl',mode='wb'))
        print("###cookie保存成功###")

        # 抢票
        seif.driver.get(target_url)

    """如果当前我已经有了cookie.pkl"""
    def get_cookie(seif):
        cookies = pickle.load(open('cookies.pkl', mode='rb'))
        for cookie in cookies:
            print(cookie)


    """登录"""
    def login(self):
        # 如果是0，则执行模拟登录
        if self.login_method == 0:
            self.driver.get(login_url)
        elif self.login_method == 1:
            # 如果当前目录下没有cookie.pkl这个文件
            if not os.path.exists('cookie.pkl'):
                # 登录一下 登录信息的记录
                self.set_cookies()
            else:
                self.driver.get(target_url)
                # 登录一下 通过seleenium传入一些信息
                self.get_cookie()


if __name__ == '__main__':
    con = Concert()
    con.login()




















